import mysql.connector


def get_con(database):
    # 异常处理机制
    try:
        con = mysql.connector.connect(
            host='127.0.0.1',
            port='3306',
            user='root',
            password='123456',
            database=database,
            charset='utf8'
        )
    except mysql.connector.Error as e:
        print(str(e))
    else:
        return con


# 编写一个查询数据库中的数据的方法
def get_sql_content(database, table_name, *args):
    """
    查询数据库中的数据的方法
    :param database: 库名
    :param table_name: 表名
    :param args: 字段名
    :return: 查询完毕之后的数据
    """
    con = get_con(database)
    sql = "select "
    if len(args) != 0:
        for i in range(0, len(args)):
            if i == len(args) - 1:
                sql += args[i]
            else:
                sql += args[i]
                sql += ','
    sql += " from " + table_name
    cursor = con.cursor()
    cursor.execute(sql)
    result = cursor.fetchall()
    cursor.close()
    con.close()
    return result

